import random

def generate_data(num=3, n=5, max_strength=100, max_k=10):
    """
    随机生成破解锁的最少时间 I 的测试数据
    :param num: 生成数据组数
    :param n: 每组锁的数量
    :param max_strength: 单个锁最大能量需求
    :param max_k: k最大值
    :return: (strength, k) 列表
    """
    data = []
    for _ in range(num):
        strength = [random.randint(1, max_strength) for _ in range(n)]
        k = random.randint(1, max_k)
        data.append((strength, k))
    return data

if __name__ == "__main__":
    for strength, k in generate_data():
        print("能量需求：", strength, "K：", k)
